#!/usr/bin/env python
"""
手动为教师分配课程
@file manual_assign_courses.py
@description 为指定的教师分配课程
@created 2024-07-11
"""

import os
import sys
import django

# 设置Django环境
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'ai_teach.settings')
django.setup()

# 导入views模块
from admin import teacher_views

def main():
    """主函数"""
    try:
        # 获取命令行参数
        if len(sys.argv) > 1:
            teacher_id = sys.argv[1]
        else:
            # 默认教师ID，如果没有提供命令行参数
            teacher_id = 25
        
        print(f"为教师 {teacher_id} 分配课程...")
        
        # 确保基础数据存在
        teacher_views.ensure_course_data_exists()
        
        # 为指定教师分配课程
        teacher_views.assign_courses_to_teacher(teacher_id)
        
        print(f"成功为教师 {teacher_id} 分配课程")
        
    except Exception as e:
        print(f"分配课程失败: {str(e)}")

if __name__ == "__main__":
    main() 